Cognitive training via virtual reality